# +----------------------------------------------------------------------
# | ChatWork智能聊天办公系统
# +----------------------------------------------------------------------
# | 软件声明: 本系统并非自由软件,未经授权任何形式的商业使用均属非法。
# | 版权保护: 任何企业和个人不允许对程序代码以任何形式任何目的复制/分发。
# | 授权要求: 如有商业使用需求,请务必先与版权所有者取得联系并获得正式授权。
# +----------------------------------------------------------------------
# | Author: ChatWork Team <2474369941@qq.com>
# +----------------------------------------------------------------------
from tortoise.queryset import Q
from hypertext import PagingResult
from common.models.agents import AgentRobotInstructModel
from apps.admin.schemas.agent.robot import instruct_schema as schema


class RobotInstructService:
    """智能体指令服务类"""

    @classmethod
    async def lists(cls, params: schema.RobotInstructSearchIn) -> PagingResult[schema.RobotInstructListVo]:
        """
        指令列表。

        Args:
            params (schema.RobotInstructSearchIn): 指令查询参数。

        Returns:
            PagingResult[schema.RobotInstructListVo]: 指令分页列表Vo。

        Author:
            zero
        """
        # 查询数据
        _where = [Q(robot_id=params.robot_id)]
        _model = AgentRobotInstructModel.filter(*_where).order_by("-sort", "-id")
        _pager = await AgentRobotInstructModel.paginate(
            model=_model,
            page_no=params.page_no,
            page_size=params.page_size
        )

        # 处理数据
        _lists = []
        for item in _pager.lists:
            _lists.append(schema.RobotInstructListVo(
                id=int(item["id"]),
                keyword=item["keyword"],
                content=item["content"],
                create_time=item["create_time"],
                update_time=item["update_time"]
            ))

        _pager.lists = _lists
        return _pager
